【CUDA+Anaconda+Pytorch的安装】 | 您所在的位置:网站首页 › nvidia 清洁安装 › 【CUDA+Anaconda+Pytorch的安装】 |
CUDA+Anaconda+Pytorch的安装
CUDA安装一、检查设备是否支持CUDA二、没有安装VS三、提示Nsight compute安装失败
Anaconda安装Pytorch安装换源
CUDA安装
一、检查设备是否支持CUDA
首先,在安装CUDA之前,请先去官网确认自己电脑的显卡是否支持安装CUDA 二、没有安装VS若电脑没有装vs就不要勾选相应组件(visual studio intergratin),装了的可以勾选 三、提示Nsight compute安装失败选择自定义安装并且取消勾选Nsight compute 安装结束后可以通过在命令行输入nvcc -V检测是否安装成功 显示该界面证明安装成功 Anaconda安装直接搜索anaconda官网下载即可(注意不要安装在C盘,因为里面要下载很多虚拟环境,会占据大量空间) Anaconda安装成功后可以通过在Anaconda Prompt输入 conda list检验是否安装成功 在Anaconda Prompt输入如下 conda create -n pytorch python=3.9.7 #创建一个新的虚拟环境命名为 pytorch conda activate pytorch #激活名为pytorch的虚拟环境Anaconda 的其他操作 conda activate base #回到真实环境 conda remove -n pytorch --all #删除名为pytorch的虚拟环境 Pytorch安装 换源因为外网下载pytorch会很慢,所以换成国内源,这里我使用清华源。 conda config --set show_channel_urls yeswindow用户看不到condrac文件,所以用上面代码生成。之后在c盘,用户文件夹里面就能找到 把文件夹内原始内容删掉,然后换成下面的内容,直接复制粘贴进去,记得保存!(用记事本打开即可) channels: - defaults show_channel_urls: true default_channels: - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ custom_channels: conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud换源完成后 进入pytorch官网查看你要安装的版本号复制代码在虚拟环境中安装即可 例 conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia安装结束后Anaconda Prompt中输入,如果没有报错,那么恭喜你,安装成功啦! conda activate pytorch python import torch |
CopyRight 2018-2019 实验室设备网 版权所有 |